﻿//------------------------------------------------------------------------------
//  此代码版权声明为全文件覆盖，如有原作者特别声明，会在下方手动补充
//  此代码版权（除特别声明外的代码）归作者本人Diego所有
//  源代码使用协议遵循本仓库的开源协议及附加协议
//  Gitee源代码仓库：https://gitee.com/diego2098/ThingsGateway
//  Github源代码仓库：https://github.com/kimdiego2098/ThingsGateway
//  使用文档：https://thingsgateway.cn/
//  QQ群：605534569
//------------------------------------------------------------------------------

namespace ThingsGateway.Gateway.Razor;

[ThingsGateway.DependencyInjection.SuppressSniffer]
public static class GatewayResourceUtil
{


    /// <summary>
    /// 构建树节点，传入的列表已经是树结构
    /// </summary>
    public static List<TreeViewItem<ChannelDeviceTreeItem>> BuildTreeItemList(this IEnumerable<ChannelRuntime> channelRuntimes, List<ChannelDeviceTreeItem> selectedItems, Microsoft.AspNetCore.Components.RenderFragment<ChannelDeviceTreeItem> render = null, TreeViewItem<ChannelDeviceTreeItem>? parent = null, List<TreeViewItem<ChannelDeviceTreeItem>> items = null)
    {
        if (channelRuntimes == null) return null;
        items ??= new();
        var trees = new List<TreeViewItem<ChannelDeviceTreeItem>>();

        //筛选插件名称
#pragma warning disable CA1851
        foreach (var pluginName in channelRuntimes.Select(a => a.PluginName).ToHashSet())
        {
            var pluginItemValue = new ChannelDeviceTreeItem() { ChannelDevicePluginType = ChannelDevicePluginTypeEnum.PluginName, PluginName = pluginName };
            var pluginItem = items.FirstOrDefault(a => a.Value.Equals(pluginItemValue));

            if (pluginItem == null)
            {
                pluginItem = new TreeViewItem<ChannelDeviceTreeItem>(pluginItemValue)
                {
                    Text = PluginInfoUtil.GetFileNameAndTypeName(pluginName).TypeName,
                    IsExpand = true,
                    Parent = parent,
                };
            }

            var channelOldItems = pluginItem.Items.ToList();
            pluginItem.Items.Clear();
            foreach (var channelRuntime in channelRuntimes.Where(a => a.PluginName == pluginName))
            {
                var channelRuntimeTreeItem = new ChannelDeviceTreeItem() { ChannelDevicePluginType = ChannelDevicePluginTypeEnum.Channel, ChannelRuntimeId = channelRuntime.Id, Id = channelRuntime.Id };

                var channelTreeItemItem = channelOldItems.FirstOrDefault(a => a.Value.Equals(channelRuntimeTreeItem));

                if (channelTreeItemItem == null)
                {
                    channelTreeItemItem = new TreeViewItem<ChannelDeviceTreeItem>(channelRuntimeTreeItem)
                    {
                        Text = channelRuntime.ToString(),
                        Parent = pluginItem,
                        IsExpand = true,
                        IsActive = selectedItems.Contains(channelRuntimeTreeItem),
                        Template = render,
                    };
                }

                var deviceOldItems = channelTreeItemItem.Items.ToList();
                channelTreeItemItem.Items.Clear();

                foreach (var keyValue in channelRuntime.ReadDeviceRuntimes.OrderBy(a => a.Value.DeviceStatus))
                {
                    var deviceRuntime = keyValue.Value;
                    var deviceRuntimeTreeItem = new ChannelDeviceTreeItem() { ChannelDevicePluginType = ChannelDevicePluginTypeEnum.Device, DeviceRuntimeId = deviceRuntime.Id, Id = deviceRuntime.Id };

                    var deviceTreeItemItem = deviceOldItems.FirstOrDefault(a => a.Value.Equals(deviceRuntimeTreeItem));

                    if (deviceTreeItemItem == null)
                    {
                        deviceTreeItemItem = new TreeViewItem<ChannelDeviceTreeItem>(deviceRuntimeTreeItem)
                        {
                            Text = keyValue.Value.Name,
                            Parent = pluginItem,
                            IsExpand = false,
                            IsActive = selectedItems.Contains(deviceRuntimeTreeItem),
                            Template = render,
                        };
                    }

                    channelTreeItemItem.Items.Add(deviceTreeItemItem);
                }

                channelTreeItemItem.Items = channelTreeItemItem.Items.OrderBy(a => a.Value.Id).ToList();
                pluginItem.Items.Add(channelTreeItemItem);
            }
            pluginItem.Items = pluginItem.Items.OrderBy(a => a.Value.Id).ToList();
            trees.Add(pluginItem);
        }
#pragma warning restore CA1851

        return trees;
    }

    /// <summary>
    /// 构建树节点，传入的列表已经是树结构
    /// </summary>
    public static List<TreeViewItem<ChannelDeviceTreeItem>> BuildTreeItemList(this Dictionary<ChannelRuntime, List<DeviceRuntime>> dict, List<ChannelDeviceTreeItem> selectedItems, Microsoft.AspNetCore.Components.RenderFragment<ChannelDeviceTreeItem> render = null, TreeViewItem<ChannelDeviceTreeItem>? parent = null, List<TreeViewItem<ChannelDeviceTreeItem>> items = null)
    {
        if (dict == null) return null;
        items ??= new();
        var trees = new List<TreeViewItem<ChannelDeviceTreeItem>>();

        //筛选插件名称

        foreach (var pluginName in dict.Select(a => a.Key.PluginName).ToHashSet())
        {
            var pluginItemValue = new ChannelDeviceTreeItem() { ChannelDevicePluginType = ChannelDevicePluginTypeEnum.PluginName, PluginName = pluginName };

            var pluginItem = items.FirstOrDefault(a => a.Value.Equals(pluginItemValue));

            if (pluginItem == null)
            {
                pluginItem = new TreeViewItem<ChannelDeviceTreeItem>(pluginItemValue)
                {
                    Text = PluginInfoUtil.GetFileNameAndTypeName(pluginName).TypeName,
                    IsExpand = true,
                    Parent = parent,
                };
            }
            var channelOldItems = pluginItem.Items.ToList();
            pluginItem.Items.Clear();

            foreach (var channelRuntime in dict.Where(a => a.Key.PluginName == pluginName))
            {
                var channelRuntimeTreeItem = new ChannelDeviceTreeItem() { ChannelDevicePluginType = ChannelDevicePluginTypeEnum.Channel, ChannelRuntimeId = channelRuntime.Key.Id, Id = channelRuntime.Key.Id };

                var channelTreeItemItem = channelOldItems.FirstOrDefault(a => a.Value.Equals(channelRuntimeTreeItem));

                if (channelTreeItemItem == null)
                {
                    channelTreeItemItem = new TreeViewItem<ChannelDeviceTreeItem>(channelRuntimeTreeItem)
                    {
                        Text = channelRuntime.ToString(),
                        Parent = pluginItem,
                        IsExpand = true,
                        IsActive = selectedItems.Contains(channelRuntimeTreeItem),
                        Template = render,
                    };
                }

                var deviceOldItems = channelTreeItemItem.Items.ToList();
                channelTreeItemItem.Items.Clear();

                foreach (var deviceRuntime in channelRuntime.Value.OrderBy(a => a.DeviceStatus))
                {
                    var deviceRuntimeTreeItem = new ChannelDeviceTreeItem() { ChannelDevicePluginType = ChannelDevicePluginTypeEnum.Device, DeviceRuntimeId = deviceRuntime.Id, Id = deviceRuntime.Id };

                    var deviceTreeItemItem = deviceOldItems.FirstOrDefault(a => a.Value.Equals(deviceRuntimeTreeItem));

                    if (deviceTreeItemItem == null)
                    {
                        deviceTreeItemItem = new TreeViewItem<ChannelDeviceTreeItem>(deviceRuntimeTreeItem)
                        {
                            Text = deviceRuntime.Name,
                            Parent = pluginItem,
                            IsExpand = false,
                            IsActive = selectedItems.Contains(deviceRuntimeTreeItem),
                            Template = render,
                        };
                    }

                    channelTreeItemItem.Items.Add(deviceTreeItemItem);
                }

                channelTreeItemItem.Items = channelTreeItemItem.Items.OrderBy(a => a.Value.Id).ToList();
                pluginItem.Items.Add(channelTreeItemItem);
            }

            pluginItem.Items = pluginItem.Items.OrderBy(a => a.Value.Id).ToList();
            trees.Add(pluginItem);
        }

        return trees;
    }
}
